package com.kitten.chatglmsdk.model;

import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;

@Data
@AllArgsConstructor
@NoArgsConstructor
@Builder
public class ImageCompletionRequest {
    private Model model;
    private String prompt;

    public String getModel() {
        return this.model.getCode();
    }
    public Model getModelEnum() {
        return this.model;
    }

    @Override
    public String toString() {
        return "ImageCompletionRequest{" +
                "model=" + model +
                ", prompt='" + prompt + '\'' +
                '}';
    }
}
